package Illness;

import Data.Data;

public class Illness {

    // contamination rate
    private int humanContamination;
    private int pigContamination;
    private int chickenContamination;
    private int duckContmination;

    private int[] elementContamination = new int[Data.nClassElement];

    // constructor for illness which affect pig and human
    public Illness(int humanCont, int pigCont) {
        this.elementContamination[0] = humanCont;
        this.elementContamination[1] = 0;
        this.elementContamination[2] = 0;
        this.elementContamination[3] = pigCont;

    }

    // constructor for illness whick affect human, chicken and duck
    public Illness(int humanCont, int chickenCont, int duckCont) {
        this.elementContamination[0] = humanCont;
        this.elementContamination[1] = chickenCont;
        this.elementContamination[2] = duckCont;
        this.elementContamination[3] = 0;
    }

    public int getHumanCont() {
        return this.humanContamination;
    }

    public int getPigCont() {
        return this.pigContamination;
    }

    public int getChickenCont() {
        return this.chickenContamination;
    }

    public int getDuckCont() {
        return this.duckContmination;
    }

    public int[] getElementCont() {
        return this.elementContamination;
    }

}
